import numpy as np
import matplotlib.pyplot as plt

# 创建一个包含-5到5的500个点的数组
x = np.linspace(-2, 2, 101)

# 使用np.heaviside()方法，将x中的每个元素与0进行比较
# 当x中的元素小于0时，结果为0；当x中的元素大于或等于0时，结果为1
gfg = np.heaviside(x, 0.5)

# 绘制图像
plt.plot(x, gfg, label='Heaviside Function')

# 添加标题和轴标签
plt.title('Heaviside Function')
plt.xlabel('x')
plt.ylabel('H(x)')

# 标记y轴上的1和0的位置，以便更清楚地看到阶跃的位置
plt.axhline(1, color='black', linewidth=0.5)
plt.axhline(0, color='black', linewidth=0.5)

# 添加图例
plt.legend()

# 显示图像
plt.show()